import torch
from torch import nn
x = torch.ones(3)
torch.save(x, 'x.pt')
